package com.ruoyi.common.utils;

import org.springframework.util.CollectionUtils;

import java.util.ArrayList;
import java.util.List;

public class MyStringUtil {
    private static List<String> data = new ArrayList<>();

    private static List<String> getString() {
        if (!CollectionUtils.isEmpty(data)) {
            return data;
        }

        for (int i = 0; i < 10; i++) {
            data.add(i + "");
        }
//        for (int i = 0; i < 26; i++) {
//            char a = 'A';
//            char b = (char) (a + i);
//            String s = String.valueOf(b);
//            data.add(s);
//        }

        return data;
    }

    /**
     * 创建A-Z  0-9的组合随机数
     *
     * @param num
     * @return
     */
    public static String createCode(int num) {
        String result = "";
        for (int i = 0; i < num; i++) {
            int index = (int) (Math.random() * getString().size());
            result += data.get(index);
        }
        return result;
    }

    public static String getCode(Integer num) {

        String code = "";
        while (code.length() < num) {
            code += (int) (Math.random() * 10);
        }

        return code;
    }
}

